feaa65
@@ -53,11 +53,7 @@
public int compare(TaskWrapper t1, TaskWrapper t2) {
     if (o1.getQueryId().equals(o2.getQueryId())) {
       // Same Query
       // Within dag priority - lower values indicate higher priority.
-      if (fri1.getWithinDagPriority() < fri2.getWithinDagPriority()) {
-        return -1;
-      } else if (fri1.getWithinDagPriority() > fri2.getWithinDagPriority()){
-        return 1;
-      }
+      return Integer.compare(fri1.getWithinDagPriority(), fri2.getWithinDagPriority());
     }
 
     if (fri1.getFirstAttemptStartTime() < fri2.getFirstAttemptStartTime()) {
